package zzztest;

import java.util.Arrays;

public class MyArrayList<T> {
    private T[] data;

    public MyArrayList() {
        data = (T[]) new Object[10];
    }

    /**
     * 向集合中添加一个元素
     *
     * 内部实现逻辑:
     * 将元素存入数组中，如果数组中已经存满，则数组需要先扩容
     * @param t
     * @return
     */
    public boolean add(T t){
        boolean flag = false;
        if(data.length==size()){
            data= Arrays.copyOf(data, data.length*2);
        }
        for (int i=0;i<data.length;i++){
            if(data[i] == null){
                data[i] = t;
                flag=true;
                break;
            }
        }
        /*if(data.length < size()){
            //扩容
            T[] newData = (T[]) new Object[data.length*2];
            //添加元素,将元素复制到新数组中
            for(int i=0;i<data.length;i++){
                newData[i] = data[i];
            }
            //实现add t
            newData[size()+1] = t;
            //赋值
            data = newData;
        }
        for (int i=0;i<data.length;i++){
            if(data[i] == null){
                set(i,t);
                //data[i] = t;
                break;
            }
        }
        //data[size()+1] = t;*/
        return flag;
    }

    /**
     * 获取指定位置上的元素
     * @param index
     * @return
     */
    public T get(int index){
        //判断index是否合法
        if(index >= 0 && index < size()){
            return data[index];
        }else {
        return null;}
    }

    /**
     * 删除指定位置上的元素，并且将被删除的元素返回
     * @param index
     * @return
     */
    public T remove(int index){
        if(index >= 0 && index < size()){
            T t = data[index];
            //实现删除
            for(int i=index;i<size()-1;i++){
                //将下一个元素赋值给当前元素
                data[i] = data[i+1];
            }
            //删除最后一个元素
            data[size()-1] = null;
            return t;
        }
        return null;
    }

    /**
     * 返回当前集合的元素个数
     * @return
     */
    public int size(){
        int count = 0;
        //遍历数组，判断元素是否为null
        for (int i=0;i<data.length;i++){
            if(data[i] != null){
                count++;
                //return i+1;
            }
        }
        return count;
    }

    /**
     * 判断集合是否包含给定元素，判断的逻辑是该元素是否与集合中某个元素equals比较为true
     * @param t
     * @return
     */
    public boolean contains(T t){
        for (int i=0;i<size();i++){
            if(data[i].equals(t)){
                return true;
            }
        }
        return false;
    }

    /**
     * 将给定元素设置到指定位置上，并将该位置原来的元素返回
     * @param index
     * @param t
     * @return
     */
    public T set(int index, T t){
        if(index >= 0 && index < size()){
            T old = data[index];
            data[index] = t;
            return old;
        }else {
            return null;
        }

    }

    /**
     * 实现toString方法，将集合每个元素包含到字符串后返回。
     * 格式如:[元素1,元素2,元素3,...]
     * @return
     */
    public String toString(){
        if(data != null){
            StringBuilder sb = new StringBuilder();
            sb.append("[");
            for(int i=0;i<size();i++){
                    sb.append(data[i]);
                    if(i < size()-1){
                        sb.append(",");
                    }

            }
            sb.append("]");
            return sb.toString();

        }
        return "[]";
    }
}